package com.bdqn.controller;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import com.bdqn.common.Constants;
import com.bdqn.common.PageSupport;
import com.bdqn.pojo.SysRole;
import com.bdqn.pojo.SysUser;
import com.bdqn.service.SysRoleService;
import com.bdqn.service.SysUserService;
import org.apache.commons.io.FilenameUtils;
import org.apache.commons.lang3.RandomUtils;
import org.apache.log4j.Logger;
import org.springframework.stereotype.Controller;
import org.springframework.ui.Model;
import org.springframework.validation.BindingResult;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpSession;
import javax.validation.Valid;
import java.io.File;
import java.io.IOException;
import java.util.List;

/**
 * UserController
 *
 * @author LILIBO
 * @since 2022-12-31
 */
@Controller
@RequestMapping("/user")
public class UserController {

    private static final Logger logger = Logger.getLogger(UserController.class);

    @Resource
    private SysUserService sysUserService;

    @Resource
    private SysRoleService sysRoleService;

    /**
     * 跳转到登录页
     */
    @GetMapping("/login")
    public String toLogin() {
        return "login"; // 跳转到登录页
    }

    /**
     * 执行登录操作
     */
    @PostMapping("/login")
    public String doLogin(Model model, HttpSession session, String account, String password) {
        // 调用Service层进行登录验证
        SysUser loginUser = sysUserService.login(account, password);
        if (loginUser == null) {
            // 将错误信息设置到作用域，并返回login页面进行提示
            model.addAttribute("error", "账号或密码错误，请重新输入");
            return "login";
        }
        session.setAttribute("USER_SESSION", loginUser);
        return "frame"; // 跳转到主页
    }

    /**
     * 执行退出操作
     */
    @GetMapping("/logout")
    public String doLogout(HttpSession session) {
        // 移除Session中的数据
        session.removeAttribute("USER_SESSION");
        // 销毁Session（会话）
        session.invalidate();
        return "login";
    }

    /**
     * 用户列表
     *
     * @return
     */
    @RequestMapping("/list") // Get/Post请求均会请求该方法
    public String list(Model model, String queryRealName, Long queryRoleId, @RequestParam(value = "pageIndex", required = false, defaultValue = "1") Integer pageIndex) {

        // 获取所有角色
        List<SysRole> roleList = sysRoleService.getAll();

        // 根据参数分页查询用户列表
        PageSupport page = new PageSupport();
        page.setCurrentPageNo(pageIndex);
        int count = sysUserService.getCount(queryRealName, queryRoleId);
        page.setTotalCount(count);
        List<SysUser> userList = sysUserService.getPage(queryRealName, queryRoleId, page);

        // 将数据设置到作用域
        model.addAttribute("queryRealName", queryRealName); // 查询条件 数据回显
        model.addAttribute("queryRoleId", queryRoleId); // 查询条件 数据回显
        model.addAttribute("pageIndex", pageIndex); // 查询条件 数据回显
        model.addAttribute("roleList", roleList); // 角色下拉框数据准备
        model.addAttribute("userList", userList); // 分页查询用户列表数据
        model.addAttribute("page", page); // 分页对象

        return "sysUser/list"; // 跳转到sysUser模块下的list页面
    }

    /**
     * 跳转到添加页面
     *
     * @return
     */
    // @GetMapping("/add")
    public String toAdd(@ModelAttribute("sysUser") SysUser sysUser, Model model) {
        // 获取所有角色
        List<SysRole> roleList = sysRoleService.getAll();
        model.addAttribute("roleList", roleList);
        return "sysUser/add";
    }

    /**
     * 执行添加保存
     *
     * @return
     */
    // @PostMapping("/add")
    public String doAdd(@Valid SysUser sysUser, BindingResult bindingResult) {
        if (bindingResult.hasErrors()) {
            logger.debug("/user/add 添加用户验证失败");
            return "sysUser/add";
        }
        return "redirect:/user/list"; // 重定向，到某个请求
    }

    /**
     * 跳转到添加页面
     */
    @GetMapping("/add")
    public String toAdd(Model model) {
        // 获取所有角色
        List<SysRole> roleList = sysRoleService.getAll();
        model.addAttribute("roleList", roleList);
        return "sysUser/add";
    }

    /**
     * 判断用户是否存在（0：账号不存在；1：账号已存在）
     */
    @ResponseBody
    @GetMapping(value = "/userExist", produces={"application/json;charset=UTF-8"})
    public String userExist(@RequestParam String account) {
        SysUser sysUser = sysUserService.getByAccount(account);
        int exist = sysUser == null ? 0 : 1;

        // 使用阿里巴巴FastJson以对象的形式组织JSON数据
        JSONObject json = new JSONObject();
        json.put("exist", exist);
        if (sysUser != null) {
            // 演示FastJson对象形式组织数据
            JSONObject jsonUser = new JSONObject();
            jsonUser.put("id", sysUser.getId());
            jsonUser.put("userName", sysUser.getRealName());
            json.put("user", jsonUser);
        }

        return json.toJSONString(); // {exist:1,user:{id:1,userName:"管理员"}}
    }

    /**
     * 添加用户（带文件上传）
     */
    @PostMapping("/add")
    public String doAdd(HttpServletRequest request, HttpSession session, SysUser sysUser, @RequestParam(value = "idPic", required = false) MultipartFile idPic, @RequestParam(value = "workPic", required = false) MultipartFile workPic) {
        // 从Session中获取当前登录用户
        SysUser loginUser = (SysUser) session.getAttribute(Constants.USER_SESSION);
        // 设置现在新添加的用户是由哪个账号创建的（填充数据库中createdUserId字段）
        sysUser.setCreatedUserId(loginUser.getId());

        // 文件上传到服务器的真实位置
        String uploadPath = session.getServletContext().getRealPath(Constants.UPLOAD_PATH);

        // 上传身份证照片
        if (idPic != null) {
            String idPicPath = multipartFileUpload(uploadPath, request, idPic);
            if (idPicPath == null) {
                String uploadFileError = (String) request.getAttribute("uploadFileError");
                request.setAttribute("idPicUploadError", uploadFileError);
                request.removeAttribute("uploadFileError"); // 临时使用，使用完毕后移除信息
                return "sysUser/add";
            }
            // 将上传后的图片路径设置到属性，最终保存到数据库
            sysUser.setIdPicPath(idPicPath);
        }

        // 上传工作证照片
        if (workPic != null) {
            String workPicPath = multipartFileUpload(uploadPath, request, workPic);
            if (workPicPath == null) {
                String uploadFileError = (String) request.getAttribute("uploadFileError");
                request.setAttribute("workPicUploadError", uploadFileError);
                request.removeAttribute("uploadFileError"); // 临时使用，使用完毕后移除信息
                return "sysUser/add";
            }
            // 将上传后的图片路径设置到属性，最终保存到数据库
            sysUser.setWorkPicPath(workPicPath);
        }

        // 将数据保存到数据库
        boolean flag = sysUserService.add(sysUser);
        logger.debug("/user/add " + flag + "\n" + sysUser);
        // 添加功能完成，跳转到列表页面
        return "redirect:/user/list"; // 使用重定向（让浏览器重新发起请求）
    }

    /**
     * 多部件文件上传（错误信息临时保存在request作用域uploadFileError属性中）
     *
     * @return 成功返回文件上传地址，返回null表示文件上传失败
     */
    public synchronized String multipartFileUpload(String uploadPath, HttpServletRequest request, MultipartFile multipartFile) {
        String uploadFilePath = null; // 文件上传地址，返回null表示文件上传失败

        // 定义文件存放目录
        logger.debug("上传文件路径：" + uploadPath);

        if (!multipartFile.isEmpty()) {
            // 获取上传的文件信息
            String originalFile = multipartFile.getOriginalFilename(); // 原文件名称
            logger.debug("原文件名称为：" + originalFile);
            String prefix = FilenameUtils.getExtension(originalFile); // 原文件后缀
            logger.debug("原文件后缀为：" + prefix);
            long fileSize = multipartFile.getSize(); // 原文件大小
            logger.debug("上传文件大小：" + fileSize);

            // 判断上传的文件是否符合要求
            if (fileSize > Constants.MAX_FILE_SIZE) { // 上传文件大小不能超过500KB
                request.setAttribute("uploadFileError", " * 上传文件大小不得超过 5MB");
                return uploadFilePath;
            }
            if (!(prefix.equalsIgnoreCase("jpg") || prefix.equalsIgnoreCase("jpeg") || prefix.equalsIgnoreCase("png") || prefix.equalsIgnoreCase("pneg"))) {
                request.setAttribute("uploadFileError", " * 上传文件格式只能是jpg、jpeg、png、pneg");
                return uploadFilePath;
            }
            // 确定文件存储名称
            String fileName = System.nanoTime() + RandomUtils.nextInt(0, 1000) + "_Personal." + prefix;
            logger.debug("新生成的文件名为：" + fileName);

            // 将上传的文件保存到目标位置
            File targetFile = new File(uploadPath);
            if (!targetFile.exists()) {
                targetFile.mkdirs(); // 如果目录不存在则生成目录
            }
            // 保存文件
            try {
                // 存储物理地址 + 生成的文件名称
                // D:/uploadFile/123456999_Personal.jpg
                multipartFile.transferTo(new File(targetFile, fileName));
            } catch (IOException e) {
                e.printStackTrace();
                request.setAttribute("uploadFileError", " * 文件上传失败！");
                return uploadFilePath;
            }
            // 文件上传完成就可以拿到地址
            uploadFilePath = File.separator + Constants.UPLOAD_PATH + File.separator + fileName;
        }

        // 返回文件上传成功后的地址，失败时返回null，并且将失败信息记录到
        return uploadFilePath;
    }

    /**
     * Ajax查看用户详情
     */
    @ResponseBody // 将方法返回内容输出到浏览器，通常处理Ajax请求
    @GetMapping("/{id}/view")
    public String view(@PathVariable("id") Integer id) {

        SysUser sysUser = sysUserService.getById(id);

        // 用户不存在
        if (sysUser == null) {
            return "nodata";
        }

        // 使用Alibaba提供的FastJSON将对象转换为Json字符串
        String jsonStr = JSON.toJSONString(sysUser);

        return jsonStr;
    }
}
